package com.NetSunWeiQin.chapter03.multithread3;

import java.io.*;
import java.net.*;
import java.util.concurrent.*;


/**
 * 使用线程池
 */
public class ThreadPoolEchoServer {
    private int port = 8000;
    private ServerSocket serverSocket;
    private ExecutorService executorService;
    private final int POOL_SIZE = 2;

    public ThreadPoolEchoServer() throws IOException {
        serverSocket = new ServerSocket(port);

//        executorService = Executors.newFixedThreadPool(  Runtime.getRuntime().availableProcessors() * POOL_SIZE);
        executorService = Executors.newFixedThreadPool(POOL_SIZE *2 );

        System.out.println(   " INFO server started.   address => "    +serverSocket.getInetAddress().getHostName()  +"\\" +   port );
    }

    public void service() {

        int count = 1;

        while (true) {
            Socket socket  ;
            try {
                socket = serverSocket.accept();
                executorService.execute(new Handler(socket ,  "线程" + count + "号"  ));
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }

    public static void main(String args[]) throws IOException {
        new ThreadPoolEchoServer().service();
    }
}

class Handler implements Runnable {


    private Socket socket;

    public Handler(Socket socket  , String  name ) {
        this.socket = socket;
    }

    private PrintWriter getWriter(Socket socket) throws IOException {
        OutputStream socketOut = socket.getOutputStream();
        return new PrintWriter(socketOut, true);
    }

    private BufferedReader getReader(Socket socket) throws IOException {
        InputStream socketIn = socket.getInputStream();
        return new BufferedReader(new InputStreamReader(socketIn));
    }

    public String echo(String msg) {
        return "echo:" + msg;
    }

    public void run() {
        try {
            System.out.println("New connection accepted " + socket.getInetAddress() + ":" + socket.getPort());
            BufferedReader br = getReader(socket);
            PrintWriter pw = getWriter(socket);

            String msg = null;
            while ((msg = br.readLine()) != null) {
                System.out.println(" INFO " + Thread.currentThread().getName() + " handle request which  from " + socket.getInetAddress() + ":" + socket.getPort() + ">" + msg);

                pw.println(echo(msg));
                if (msg.equals("bye"))
                    break;
            }
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            try {
                if (socket != null) socket.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }
}
